package org.sun.consumer.service;

import com.alibaba.csp.sentinel.Entry;
import com.alibaba.csp.sentinel.SphO;
import com.alibaba.csp.sentinel.SphU;
import com.alibaba.csp.sentinel.slots.block.BlockException;
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;

import java.util.ArrayList;
import java.util.List;

/**
 * sentinal  默认情况下。规则是配置在节点的内存里的。针对的是单机的流量控制。也就是服务所在的服务器的流量控制。
 * 在生产环境下使用，要考虑sentinel 规则配置信息持久化的问题，需要进行改造。
 *
 */
public class SentinalService {
    public static void main(String[] args) {
        // 配置规则.
        initFlowRules();
        //https://sentinelguard.io/zh-cn/docs/basic-implementation.html
        while (true) {
            // 1.5.0 版本开始可以直接利用 try-with-resources 特性
            //此种写法Entry类中实现了autoclose接口，autoclose中的操作就会在try结束后执行，所以不用再写finally。
            try (Entry entry = SphU.entry("HelloWorld")) {
                // 被保护的逻辑 可以是任何逻辑
                System.out.println("hello world");
            } catch (BlockException ex) {
                // 处理被流控的逻辑
                System.out.println("blocked!");
            }
        }



    }

    public void sph0() {
        // 资源名可使用任意有业务语义的字符串
        //SphO 提供 if-else 风格的 API
        if (SphO.entry("自定义资源名")) {
            // 务必保证finally会被执行
            try {
                /**
                 * 被保护的业务逻辑
                 */
            } finally {
                SphO.exit();
            }
        } else {
            // 资源访问阻止，被限流或被降级
            // 进行相应的处理操作
        }
    }

    /**
     * 在生产环境中使用sentinal 需要对其进行改造：
     * https://github.com/alibaba/Sentinel/wiki/%E5%9C%A8%E7%94%9F%E4%BA%A7%E7%8E%AF%E5%A2%83%E4%B8%AD%E4%BD%BF%E7%94%A8-Sentinel
     */

    private static void initFlowRules(){
        List<FlowRule> rules = new ArrayList<>();
        FlowRule rule = new FlowRule();

        rule.setResource("HelloWorld");
        rule.setGrade(RuleConstant.FLOW_GRADE_QPS);
        // Set limit QPS to 20.
        rule.setCount(20);

        rules.add(rule);
        FlowRuleManager.loadRules(rules);
    }
}
